import tensorflow as tf


def lr_scheduler(epoch,lr):
    if epoch<100:
        pass
    elif epoch < 200:
        lr = max(lr*tf.math.exp(-0.1),1e-4)
    else:
        lr = max(lr*tf.math.exp(-0.1),1e-5)
    
    return lr
    